package com.example.demo.controller;

import com.example.demo.common.ResultAjax;
import com.example.demo.common.SessionUtils;
import com.example.demo.model.Articleinfo;
import com.example.demo.model.Userinfo;
import com.example.demo.model.vo.UserinfoVO;
import com.example.demo.service.ArticleService;
import com.example.demo.service.UserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletRequest;
import java.util.HashMap;
import java.util.List;
import java.util.concurrent.ExecutionException;

import java.util.concurrent.FutureTask;

@RestController
@RequestMapping("/article")
public class ArticleController {
    private static final int _DESC_LENGTH = 120; // 文章简介的长度
    @Autowired
    private ArticleService articleService;
    @Autowired
    private UserService userService;
    @Autowired
    private ThreadPoolTaskExecutor taskExecutor;

    /**
     * 得到当前登录用户的文章列表
     * @param request
     * @return
     */
    @RequestMapping("/mylist")
    public ResultAjax myList(HttpServletRequest request){
        // 1. 得到当前登录用户
        Userinfo userinfo = SessionUtils.getUser(request);
        if(userinfo == null){
            return ResultAjax.fail(-1,"请先登录!");
        }
        // 2. 根据用户 id 查询此用户发表的所有文章
        List<Articleinfo> list = articleService.getListByUid(userinfo.getId());
        //处理 list -> 将文章正文变成简介
        if(list != null && list.size() > 0){
            // 并行处理 list 集合
            list.stream().parallel().forEach((art) ->{
                    if(art.getContent().length() > _DESC_LENGTH){
                        art.setContent(art.getContent().substring(0,_DESC_LENGTH)+"...");
                    }
            });
        }

        // 3. 返回数据给前端
        return ResultAjax.success(list);
    }

    /**
     * 删除文章
     * @param aid
     * @param request
     * @return
     */
    @RequestMapping("/del")
    public ResultAjax del(Integer aid, HttpServletRequest request){
        // 1. 参数效验
        if(aid == null || aid <= 0){
            return ResultAjax.fail(-1,"参数错误！");
        }
        // 2. 得到当前登录的用户
        Userinfo userinfo = SessionUtils.getUser(request);
        if(userinfo == null){
            return ResultAjax.fail(-2,"请先登录!");
        }
        // 3. 判断文章的归属人 + 删除操作
        int result = articleService.del(aid,userinfo.getId());
        // 4. 将结果返回给前端
        return ResultAjax.success(result);
    }

    /**
     * 添加文章
     * @param articleinfo
     * @param request
     * @return
     */
    @RequestMapping("/add")
    public ResultAjax add(Articleinfo articleinfo, HttpServletRequest request){
        // 1.效验参数
        if(articleinfo == null || !StringUtils.hasLength(articleinfo.getTitle()) ||
        !StringUtils.hasLength(articleinfo.getContent())){
            return ResultAjax.fail(-1,"非法参数！");
        }
        // 2.组装数据
        Userinfo userinfo = SessionUtils.getUser(request);
        if(userinfo == null){
            return ResultAjax.fail(-2,"请先登录！");
        }
        articleinfo.setUid(userinfo.getId());
        // 3. 将数据入库
        int result = articleService.add(articleinfo);
        // 4. 将结果返回给前端
        return ResultAjax.success(result);
    }


    /**
     * 查询自己发表的文章详情
     * @param aid
     * @param request
     * @return
     */
    @RequestMapping("/update_init")
    public ResultAjax updateInit(Integer aid, HttpServletRequest request){
        // 1. 参数效验
        if(aid == null || aid <= 0){
            return ResultAjax.fail(-1,"非法参数！");
        }
        // 2. 得到当前登录用户 id
        Userinfo userinfo = SessionUtils.getUser(request);
        if(userinfo == null){
            return ResultAjax.fail(-2,"请先登录");
        }
        // 3. 查询文章并效验权限 where id = #{aid} and uid = #{uid}
        Articleinfo articleinfo = articleService.getArticleByIdAndUid(
                aid, userinfo.getId()
        );
        // 4. 将结果返回给前端
        return ResultAjax.success(articleinfo);
    }

    /**
     * 修改操作
     * @param articleinfo
     * @param request
     * @return
     */
    @RequestMapping("/update")
    public ResultAjax update(Articleinfo articleinfo, HttpServletRequest request){
        // 1. 效验参数
        if(articleinfo == null ||
        !StringUtils.hasLength(articleinfo.getTitle()) ||
        !StringUtils.hasLength(articleinfo.getContent()) ||
        articleinfo.getId() == 0){
            return ResultAjax.fail(-1,"非法参数！");
        }
        // 2. 获取登录用户
        Userinfo userinfo = SessionUtils.getUser(request);
        if (userinfo == null){
            return ResultAjax.fail(-2,"请先登录！");
        }
        articleinfo.setUid(userinfo.getId());

        // 3. 修改文章，并效验归属人
        int result = articleService.update(articleinfo);
        // 4. 返回结果
        return ResultAjax.success(result);
    }

    /**
     * 查询文章详情页
     * @param aid
     * @return
     * @throws ExecutionException
     * @throws InterruptedException
     */
    @RequestMapping("/detail")
    public ResultAjax detail(Integer aid) throws ExecutionException, InterruptedException {
        // 1. 参数效验
        if(aid == null || aid <= 0){
            return ResultAjax.fail(-1,"参数有误！");
        }
        // 2. 查询文章详情
        Articleinfo articleinfo = articleService.getDetail(aid);
        if(articleinfo == null || articleinfo.getId() <= 0){
            return ResultAjax.fail(-1,"非法参数！");
        }
        // 3. 根据 uid 查询用户的详情
        FutureTask<UserinfoVO> userTask = new FutureTask(()->{
            return userService.getUserById(articleinfo.getUid());
        });
        taskExecutor.submit(userTask);
        // 4. 根据 uid 查询用户发表的总文章数
        FutureTask<Integer> artCountTask = new FutureTask<>(()->{
            return articleService.getArtCountByUid(articleinfo.getUid());
        });
        taskExecutor.submit(artCountTask);
        // 5. 组装数据
        UserinfoVO userinfoVO = userTask.get(); // 等待任务（线程池）执行完成
        int artCount = artCountTask.get(); // 等待任务（线程池）执行完成
        userinfoVO.setArtCount(artCount);
        HashMap<String,Object> result = new HashMap<>();
        result.put("user",userinfoVO);
        result.put("art",articleinfo);
        // 6. 返回结果给前端
        return ResultAjax.success(result);
    }

    /**
     * 阅读量更新
     * @param aid
     * @return
     */
    @RequestMapping("/increment_rcount")
    public ResultAjax incrementRCount(Integer aid){
        // 1. 参数效验
        if(aid == null || aid <= 0){
            return ResultAjax.fail(-1,"参数有误！");
        }
        // 2. 更新数据库
        int result = articleService.incrementRCount(aid);
        // 3. 返回结果
        return ResultAjax.success(result);
    }


    /**
     * 查询分页功能
     * @param psize
     * @param pindex
     * @return
     * @throws ExecutionException
     * @throws InterruptedException
     */
    @RequestMapping("/getlistbypage")
    public ResultAjax getListByPage(Integer psize, Integer pindex) throws ExecutionException, InterruptedException {
        // 1. 参数矫正
        if(pindex == null || pindex < 1){
            pindex = 1;
        }
        if(psize == null || psize < 1){
            psize = 2;
        }
        // 2. 并发进行文章列表和总页数的查询
        // 2.1 查询分页列表数据
        int finalOffset = psize * (pindex - 1); // 分页公式
        int finalPsize = psize;
        FutureTask<List<Articleinfo>> listTask = new FutureTask<>(()->{
            return articleService.getListByPage(finalPsize,finalOffset);
        });
        // 2.2 查找总页数
        FutureTask<Integer> sizeTask = new FutureTask<>(()->{
           // 总条数
           int totalCount = articleService.getCount();
           double sizeTemp = (totalCount * 1.0) / (finalPsize * 1.0);
           return (int)Math.ceil(sizeTemp);
        });
        taskExecutor.submit(listTask);
        taskExecutor.submit(sizeTask);
        // 3. 组装数据
        List<Articleinfo> list = listTask.get();
        int size = sizeTask.get();
        HashMap<String,Object> map = new HashMap<>();
        map.put("list",list);
        map.put("size",size);
        // 4. 将结果返回给前端
        return ResultAjax.success(map);

    }

}
